package com.farsunset.webhook.feign.request;

import com.fasterxml.jackson.annotation.JsonProperty;

public class OpenAIRequest {

    private final String model = "text-davinci-003";

    @JsonProperty("max_tokens")
    private final int maxTokens = 2048;

    @JsonProperty("frequency_penalty")
    private final int frequencyPenalty = 0;

    @JsonProperty("presence_penalty")
    private final int presencePenalty = 0;

    private final float temperature = 0.8F;

    private String prompt;



    public String getModel() {
        return model;
    }

    public int getMaxTokens() {
        return maxTokens;
    }

    public int getFrequencyPenalty() {
        return frequencyPenalty;
    }

    public int getPresencePenalty() {
        return presencePenalty;
    }

    public float getTemperature() {
        return temperature;
    }

    public String getPrompt() {
        return prompt;
    }

    public void setPrompt(String prompt) {
        this.prompt = prompt;
    }
}
